package com.w3c.pragmatic.utils;

import com.w3c.pragmatic.common.enums.RandomEnum;

import java.util.Random;

/**
 * 生产4位随机数，不够则补0
 *
 * @author wangdong
 * @since 2024/5/10 10:29
 */
public class RandomUtil {

  /**
   * 生成随机数，不够的用0补齐
   *
   * @param scale 随机数长度(2~9有效，其他默认2位)
   * @return 结果
   */
  public static String genRandom(int scale) {
    RandomEnum randomEnum = RandomEnum.getEnumByType(scale);
    int random = new Random().nextInt(randomEnum.getRandomRange());
    return String.format(randomEnum.getScaleFormat(), random);
  }

  /**
   * 生成随机数，不够的用0补齐
   *
   * @param scale 随机数长度(2~9有效，其他默认2位)
   * @return 结果
   */
  public static String genRandom(int scale, String prefix) {
    RandomEnum randomEnum = RandomEnum.getEnumByType(scale);
    int random = new Random().nextInt(randomEnum.getRandomRange());
    return String.format(prefix + randomEnum.getScaleFormat(), random);
  }
}
